package org.layuiframework.stereotype.util;

import org.apache.tomcat.util.codec.binary.Base64;
import org.layuiframework.stereotype.security.bean.AbstractKeyPair;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import javax.crypto.Cipher;
import java.security.KeyFactory;
import java.security.interfaces.RSAPrivateKey;
import java.security.interfaces.RSAPublicKey;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;

@Component
public class RSAUtils {

	@Resource
	AbstractKeyPair abstractKeyPair;

	/**
	 * RSA公钥加密 
	 *  
	 * @param str 
	 *            加密字符串
	 *            公钥
	 * @return 密文 
	 * @throws Exception 
	 *             加密过程中的异常信息 
	 */  
	public String encrypt( String str ) throws Exception{
		//base64编码的公钥
		byte[] decoded = Base64.decodeBase64(abstractKeyPair.getPublicKey());
		RSAPublicKey pubKey = (RSAPublicKey) KeyFactory.getInstance("RSA").generatePublic(new X509EncodedKeySpec(decoded));
		//RSA加密
		Cipher cipher = Cipher.getInstance("RSA");
		cipher.init(Cipher.ENCRYPT_MODE, pubKey);
		String outStr = Base64.encodeBase64String(cipher.doFinal(str.getBytes("UTF-8")));
		return outStr;
	}
	
	/** 
	 * RSA私钥解密
	 *  
	 * @param str 
	 *            加密字符串
	 * @return 铭文
	 * @throws Exception 
	 *             解密过程中的异常信息 
	 */  
	public  String decrypt(String str) {
		try {
			//64位解码加密后的字符串
			byte[] inputByte = Base64.decodeBase64(str.getBytes("UTF-8"));
			//base64编码的私钥
			byte[] decoded = Base64.decodeBase64(abstractKeyPair.getPrivateKey());
			RSAPrivateKey priKey = (RSAPrivateKey) KeyFactory.getInstance("RSA").generatePrivate(new PKCS8EncodedKeySpec(decoded));  
			//RSA解密
			Cipher cipher = Cipher.getInstance("RSA");
			cipher.init(Cipher.DECRYPT_MODE, priKey);
			String outStr = new String(cipher.doFinal(inputByte));
			return outStr;
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		return null;
	}

	public static void main(String[] args) throws Exception {
//		String presentedPassword = "Qw123456@";
//		// 这里是传入校验的密码
//		System.out.println(encrypt(presentedPassword));
//		PasswordEncoder encode  = new BCryptPasswordEncoder();
//		String bc = encode.encode(presentedPassword);
//		// 这里是数据库里面的密码
//		System.out.println("数据库里面保存的密码..........");
//		System.out.println(bc);
	}
}
